import os
from tensorflow.python import pywrap_tensorflow

print(os.path.abspath('./checkpoint'))
reader = pywrap_tensorflow.NewCheckpointReader(os.path.abspath('./checkpoint'))
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print('tensor_name:', key)
